#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
评论处理工具使用示例
"""

from comment_processor import CommentProcessor
import pandas as pd

def example_single_comment():
    """示例：处理单条评论"""
    print("=== 单条评论处理示例 ===")
    
    processor = CommentProcessor()
    
    # 测试评论
    test_comments = [
        "欢潮转珠棒😀很好用，，，，",
        "这个商品很差😡，质量不好",
        "在淘宝买的，包装精美❤️",
        "欢潮转珠棒😀很好用，，，，",  # 重复评论
        "质量不错，推荐购买👍",
        "这个商品***很好用",
        "包装★精美，物流快",
        "服务☆很好，推荐♥购买",
        "这个商品*很好*，值得购买",
        "质量****不错，性价比高",
        "掌柜服务很好",
        "掌柜态度不错，回复很快",
        "掌柜很专业，推荐购买"
    ]
    
    for i, comment in enumerate(test_comments, 1):
        print(f"\n评论 {i}: {comment}")
        
        processed_comment, info = processor.process_comment(comment)
        
        print(f"处理后: {processed_comment}")
        print(f"是否差评: {info['is_negative']}")
        print(f"是否重复: {info['is_duplicate']}")
        print(f"原始长度: {info['original_length']}")
        print(f"处理后长度: {info['processed_length']}")

def example_create_sample_data():
    """示例：创建示例数据文件"""
    print("\n=== 创建示例数据文件 ===")
    
    # 创建示例数据
    sample_data = {
        '评论': [
            "欢潮转珠棒😀很好用，，，，",
            "这个商品很差😡，质量不好",
            "在淘宝买的，包装精美❤️",
            "欢潮转珠棒😀很好用，，，，",  # 重复评论
            "质量不错，推荐购买👍",
            "天猫旗舰店服务很好😊",
            "这个商品很差，退货了😠",
            "包装很好，物流快🚚",
            "欢潮转珠棒😀很好用，，，，",  # 再次重复
            "性价比很高，值得购买💯",
            "这个商品***很好用",
            "包装★精美，物流快",
            "服务☆很好，推荐♥购买",
            "这个商品*很好*，值得购买",
            "质量****不错，性价比高",
            "掌柜服务很好",
            "掌柜态度不错，回复很快",
            "掌柜很专业，推荐购买"
        ],
        '评分': [5, 1, 4, 5, 5, 4, 1, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5],
        '购买时间': ['2024-01-01', '2024-01-02', '2024-01-03', '2024-01-04', '2024-01-05',
                   '2024-01-06', '2024-01-07', '2024-01-08', '2024-01-09', '2024-01-10',
                   '2024-01-11', '2024-01-12', '2024-01-13', '2024-01-14', '2024-01-15',
                   '2024-01-16', '2024-01-17', '2024-01-18']
    }
    
    # 创建DataFrame
    df = pd.DataFrame(sample_data)
    
    # 保存到comments文件夹
    import os
    os.makedirs('comments', exist_ok=True)
    df.to_excel('comments/sample_comments.xlsx', index=False)
    
    print("示例数据文件已创建: comments/sample_comments.xlsx")

def example_process_file():
    """示例：处理Excel文件"""
    print("\n=== 处理Excel文件示例 ===")
    
    processor = CommentProcessor()
    
    # 处理示例文件
    try:
        df = processor.process_excel_file('comments/sample_comments.xlsx')
        
        print("处理完成！结果预览：")
        print(df[['评论', '评论_processed', '评论_is_negative', '评论_is_duplicate']].head(10))
        
        # 统计信息
        total_comments = len(df)
        negative_comments = df['评论_is_negative'].sum()
        duplicate_comments = df['评论_is_duplicate'].sum()
        
        print(f"\n统计信息:")
        print(f"总评论数: {total_comments}")
        print(f"差评数: {negative_comments}")
        print(f"重复评论数: {duplicate_comments}")
        print(f"有效评论数: {total_comments - negative_comments - duplicate_comments}")
        
    except Exception as e:
        print(f"处理文件时出错: {e}")

def main():
    """主函数"""
    print("评论处理工具使用示例")
    print("=" * 50)
    
    # 1. 单条评论处理示例
    example_single_comment()
    
    # 2. 创建示例数据
    example_create_sample_data()
    
    # 3. 处理Excel文件示例
    example_process_file()
    
    print("\n" + "=" * 50)
    print("示例运行完成！")

if __name__ == "__main__":
    main() 